patch_size = 1